dfs and similar dp graphs shortest paths *2400

Please click on ads to support us..

Python Code:

from collections import deque
import sys, os, io
input = io.BytesIO(os.read(0, os.fstat(0).st_size)).readline

def f(u, v):
    return u * g + v

n, m = map(int, input().split())
d = list(map(int, input().split()))
d.sort()
g, r = map(int, input().split())
inf = pow(10, 9) + 1
l0, r0 = [inf] * m, [inf] * m
for i in range(m - 1):
    d0 = d[i + 1] - d[i]
    l0[i + 1], r0[i] = d0, d0
dp = [inf] * (m * g)
dp[0] = 0
q = deque([0])
c = 0
while q:
    v = q.popleft()
    i, j = divmod(v, g)
    for ni, nj in [(i - 1, j + l0[i]), (i + 1, j + r0[i])]:
        u = f(ni, nj % g)
        if nj < g and dp[u] == inf:
            dp[u] = dp[v]
            q.appendleft(u)
        elif nj == g and dp[u] == inf:
            dp[u] = dp[v] + 1
            q.append(u)
ans = inf
for i in range(m - 1, -1, -1):
    if d[i] + g < n:
        break
    ans = min(ans, dp[f(i, 0)] * (g + r) + n - d[i])
ans = (ans + 1) % (inf + 1) - 1
print(ans)


Comments

Submit
0 Comments
More Questions

1466C - Canine poetry
74A - Room Leader
1333D - Challenges in school №41
1475B - New Year's Number
461A - Appleman and Toastman
320B - Ping-Pong (Easy Version)
948A - Protect Sheep
387A - George and Sleep
53A - Autocomplete
1729G - Cut Substrings
805B - 3-palindrome
805C - Find Amir
676C - Vasya and String
1042B - Vitamins
1729F - Kirei and the Linear Function
25D - Roads not only in Berland
1694A - Creep
659F - Polycarp and Hay
1040A - Palindrome Dance
372A - Counting Kangaroos is Fun
1396B - Stoned Game
16A - Flag
1056A - Determine Line
670B - Game of Robots
1418C - Mortal Kombat Tower
1382B - Sequential Nim
1272C - Yet Another Broken Keyboard
808A - Lucky Year
1245A - Good ol' Numbers Coloring
58B - Coins